package cn.ljy.mes.utils.websocket;

import cn.ljy.mes.web.websocket.config.WebSocketSessionManager;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;

import java.io.IOException;
import java.util.Map;

/**
 * 自定义websocket用于发送信息到客户端的工具类
 */
public class WebsocketOfSendMessageUtil {
    /**
     * 群发消息给所有客户端，设计为供所有方法调用
     * @param toSession 要发送到的客户端
     * @param message 消息内容
     */
    public static void sendMessageToOne(WebSocketSession toSession, String message) throws IOException {
        toSession.sendMessage(new TextMessage(message));
    }


    /**
     * 群发消息给所有客户端，设计为供所有方法调用
     * @param message 消息内容
     */
    public static void sendMessageToAll(String message) throws IOException {
        for (Map.Entry<String, WebSocketSession> sessionEntry : WebSocketSessionManager.getAllSessions().entrySet()) {
            WebSocketSession toSession = sessionEntry.getValue();
            //向客户端发送信息
            sendMessageToOne(toSession,message);
        }
    }

    /**
     * 从某个客户端群发消息给其他客户端
     * @param fromSession 发送消息的客户端
     * @param message 消息内容
     */
    public static void sendMessageToManyExcludeSelf(WebSocketSession fromSession,String message) throws IOException {
        for (Map.Entry<String, WebSocketSession> sessionEntry : WebSocketSessionManager.getAllSessions().entrySet()) {
            WebSocketSession toSession = sessionEntry.getValue();
            // 排除掉发送信息的客户端
            if (!fromSession.getId().equals(toSession.getId())) {
                //向客户端发送信息
                sendMessageToOne(toSession,message);
            }
        }
    }
}
